import { Injectable } from '@nestjs/common';
import { getAgent } from './utils/model';
import { data } from './utils/data';

@Injectable()
export class GisService {
  async getInfo(body: { message: string }) {
    console.log(body);
    const agent = await getAgent();
    const response = await agent.invoke({
      messages: [
        { type: 'system', content: '使用中文回答' },
        {
          type: 'system',
          content: [
            { type: 'text', text: '数据源数据：' + JSON.stringify(data) },
            {
              type: 'text',
              text: `
                url: 3D图,
                lng: 空间坐标,
                lat: 空间坐标,
                x: 坐标角度,
                d:房间号,
                height: 距离地面高度,
                user:户主
              `,
            },
          ],
        },

        {
          type: 'user',
          content: 'D1所在的位置是哪里,那里的天气如何',
        },
      ],
    });

    const message = response.messages.at(-1)?.content;

    return {
      code: 200,
      data: {
        data: message,
      },
    };
  }
}
